import util.tool as tool
from lib.llm import invoke

summarys = tool.read_json("data/摘要分隔.json")


system_template = """
你是一个专业的Markdown解析器，任务是将用户提供的Markdown文本转换为结构化的JSON数据，适合网页前端渲染。

规则要求：
1. 输出必须为严格JSON格式，禁止包含任何额外解释，包括markdown的```json```
2. 类型选择必须合理匹配内容特征
3. 保持专业客观的解析风格
4. 优先用图表类型进行匹配

图表类型参考：
- 柱状图：含数值比较的表格数据
- 折线图：时间序列或连续数据
- pieChat：百分比构成数据
- 散点图：双变量关联数据
- 雷达图：多维指标体系
- count：总数统计
- list：无序/有序列表内容
- text：文本段落文字内容


输出结果：
{{
    "title":"示例标题",
    "type":"类型",
    "data":"提取到的数据"
}}
"""

results = []
for summary in summarys:
    result = invoke(
        system_template=system_template,
        user_template="{summary}",
        input_data={"summary": summary},
    )

    content = tool.extract_json_from_str(result)
    if content:
        results.append(content)
        tool.save_json("data/分段摘要转JSON.json", results)


# content = invoke(
#     system_template=system_template,
#     user_template="{summary}",
#     input_data={"summary": summarys[0]},
# )
# print("content", content)
